/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package com.youan.log.modules.audit.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.youan.log.modules.audit.entity.AttackedAttackerIpDetail;
import com.youan.log.modules.audit.excelexportentity.AttackedIpDetailExcel;
import com.youan.log.modules.audit.excelexportentity.AttackedIpDetailWithAttackerIpExcel;
import com.youan.log.modules.audit.mapper.AttackedAttackerIpDetailMapper;
import com.youan.log.modules.audit.service.IAttackedAttackerIpDetailService;
import com.youan.log.modules.audit.vo.AttackedAttackerIpDetailVO;
import com.youan.log.util.ExcelUtils;
import org.apache.commons.lang3.StringUtils;
import org.springblade.core.tool.utils.BeanUtil;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletResponse;
import java.util.Arrays;
import java.util.List;

/**
 *  服务实现类
 *
 * @author BladeX
 * @since 2021-04-01
 */
@Service
public class AttackedAttackerIpDetailServiceImpl extends ServiceImpl<AttackedAttackerIpDetailMapper, AttackedAttackerIpDetail> implements IAttackedAttackerIpDetailService {

	@Override
	public IPage<AttackedAttackerIpDetailVO> selectAttackedAttackerIpDetailPage(IPage<AttackedAttackerIpDetailVO> page, AttackedAttackerIpDetailVO attackedAttackerIpDetail) {
		if(StringUtils.isNotBlank(attackedAttackerIpDetail.getIps())) {
			attackedAttackerIpDetail.setIpList(Arrays.asList(attackedAttackerIpDetail.getIps().split(",")));
		}
		return page.setRecords(baseMapper.selectAttackedDetailPage(page, attackedAttackerIpDetail));
	}

	@Override
	public void exportAttackedIpDetailExcel(AttackedAttackerIpDetailVO attackedAttackerIpDetail, HttpServletResponse response) {
		List<AttackedAttackerIpDetail> result = baseMapper.selectAttackedDetail(attackedAttackerIpDetail);
		String filePath;
		if(StringUtils.isNotBlank(attackedAttackerIpDetail.getGroupBy())) {
			List<AttackedIpDetailWithAttackerIpExcel> attackedIpDetailExcelList = BeanUtil.copy(result, AttackedIpDetailWithAttackerIpExcel.class);
			filePath = ExcelUtils.wirte2Excel(AttackedIpDetailWithAttackerIpExcel.class, attackedIpDetailExcelList);
		}else {
			List<AttackedIpDetailExcel> attackedIpDetailExcelList = BeanUtil.copy(result, AttackedIpDetailExcel.class);
			filePath = ExcelUtils.wirte2Excel(AttackedIpDetailExcel.class, attackedIpDetailExcelList);
		}
		ExcelUtils.exportExcel(filePath, response, true);
	}

	@Override
	public List<AttackedAttackerIpDetail> attackDetail(AttackedAttackerIpDetail attackedAttackerIpDetail) throws Exception {
		if(StringUtils.isBlank(attackedAttackerIpDetail.getAttackedIp()) && StringUtils.isBlank(attackedAttackerIpDetail.getAttackerIp())) {
			throw new Exception("IP信息为空，无法获取对应攻击者详情");
		}
		LambdaQueryWrapper<AttackedAttackerIpDetail> attackedAttackerIpDetailLqw = new LambdaQueryWrapper<>();
		if(StringUtils.isNotBlank(attackedAttackerIpDetail.getAttackedIp())) {
			attackedAttackerIpDetailLqw.eq(AttackedAttackerIpDetail::getAttackedIp, attackedAttackerIpDetail.getAttackedIp());
		}
		if(StringUtils.isNotBlank(attackedAttackerIpDetail.getAttackerIp())) {
			attackedAttackerIpDetailLqw.eq(AttackedAttackerIpDetail::getAttackerIp, attackedAttackerIpDetail.getAttackerIp());
		}
		return list(attackedAttackerIpDetailLqw);
	}

}
